package com.sys.ai.vo;

import com.alibaba.fastjson2.annotation.JSONField;
import lombok.Data;
import lombok.experimental.Accessors;

import java.io.Serializable;
import java.util.List;

/**
 * @author LGY
 * Create by 2025/5/16 15:14
 */
@Data
@Accessors(chain = true)
public class EmbeddingsVO implements Serializable {

    private String model;

    private List<EmbeddingData> data;

    private EmbeddingUsage usage;

    @Data
    @Accessors(chain = true)
    public static class EmbeddingData implements Serializable {
        private String object;

        private Float[] embedding;

        private Integer index;
    }

    @Data
    @Accessors(chain = true)
    public static class EmbeddingUsage implements Serializable {

        @JSONField(name = "prompt_tokens")
        private Integer promptTokens;

        @JSONField(name = "completion_tokens")
        private Integer completionTokens;

        @JSONField(name = "total_tokens")
        private Integer totalTokens;

    }

}
